package cn.elead.chaos.sso.server.web.system.controller;

import java.util.List;

import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;

import cn.elead.chaos.excel.ExcelUtil;

import cn.elead.chaos.core.enums.BusinessType;
import cn.elead.chaos.framework.monitor.operlog.Log;
import cn.elead.chaos.framework.web.global.ApiAssert;
import cn.elead.chaos.framework.web.global.ErrorCodeEnum;
import cn.elead.chaos.mybatils.web.controller.BaseController;
import cn.elead.chaos.mybatils.web.page.TableDataInfo;
import cn.elead.chaos.shiro.util.ShiroUtils;
import cn.elead.chaos.sso.server.constants.UserConstants;
import cn.elead.chaos.sso.server.web.system.domain.SysRole;
import cn.elead.chaos.sso.server.web.system.domain.SysUser;
import cn.elead.chaos.sso.server.web.system.domain.SysUserRole;
import cn.elead.chaos.sso.server.web.system.service.ISysRoleService;
import cn.elead.chaos.sso.server.web.system.service.ISysUserService;

/**
 * 角色信息
 * 
 * @author ruoyi
 */
@Controller
@RequestMapping("/system/role")
public class SysRoleController extends BaseController {
	private String prefix = "system/role";

	@Autowired
	private ISysRoleService roleService;

	@Autowired
	private ISysUserService userService;

	@RequiresPermissions("system:role:view")
	@GetMapping()
	public String role() {
		return prefix + "/role";
	}

	@RequiresPermissions("system:role:list")
	@PostMapping("/list")
	@ResponseBody
	public TableDataInfo<SysRole> list(SysRole role) {
		startPage();
		List<SysRole> list = roleService.selectRoleList(role);
		return getDataTable(list);
	}

	@ResponseBody
	@PostMapping("/export")
	@RequiresPermissions("system:role:export")
	@Log(title = "角色管理", businessType = BusinessType.EXPORT)
	public String export(SysRole role) {
		List<SysRole> list = roleService.selectRoleList(role);
		ExcelUtil<SysRole> util = new ExcelUtil<SysRole>(SysRole.class);
		return util.exportExcel(list, "角色数据");
	}

	/**
	 * 新增角色
	 */
	@GetMapping("/add")
	public String add() {
		return prefix + "/add";
	}

	/**
	 * 新增保存角色
	 */
	@ResponseBody
	@PostMapping("/add")
	@RequiresPermissions("system:role:add")
	@Log(title = "角色管理", businessType = BusinessType.INSERT)
	public void addSave(@Validated SysRole role) {
		if (UserConstants.ROLE_NAME_NOT_UNIQUE == roleService.checkRoleNameUnique(role)) {
			ApiAssert.failure(ErrorCodeEnum.BAD_REQUEST.convert("新增角色'" + role.getRoleName() + "'失败，角色名称已存在"));
		} else if (UserConstants.ROLE_KEY_NOT_UNIQUE == roleService.checkRoleKeyUnique(role)) {
			ApiAssert.failure(ErrorCodeEnum.BAD_REQUEST.convert("新增角色'" + role.getRoleName() + "'失败，角色权限已存在"));
		}
		role.setCreateBy(ShiroUtils.getLoginName());
		ShiroUtils.clearCachedAuthorizationInfo();
		roleService.insertRole(role);

	}

	/**
	 * 修改角色
	 */
	@GetMapping("/edit/{roleId}")
	public String edit(@PathVariable("roleId") Long roleId, ModelMap mmap) {
		mmap.put("role", roleService.selectRoleById(roleId));
		return prefix + "/edit";
	}

	/**
	 * 修改保存角色
	 */
	@ResponseBody
	@PostMapping("/edit")
	@RequiresPermissions("system:role:edit")
	@Log(title = "角色管理", businessType = BusinessType.UPDATE)
	public void editSave(@Validated SysRole role) {
		roleService.checkRoleAllowed(role);
		if (UserConstants.ROLE_NAME_NOT_UNIQUE == roleService.checkRoleNameUnique(role)) {
			ApiAssert.failure(ErrorCodeEnum.BAD_REQUEST.convert("修改角色'" + role.getRoleName() + "'失败，角色名称已存在"));
		} else if (UserConstants.ROLE_KEY_NOT_UNIQUE == roleService.checkRoleKeyUnique(role)) {
			ApiAssert.failure(ErrorCodeEnum.BAD_REQUEST.convert("修改角色'" + role.getRoleName() + "'失败，角色权限已存在"));
		}
		role.setUpdateBy(ShiroUtils.getLoginName());
		ShiroUtils.clearCachedAuthorizationInfo();
		roleService.updateRole(role);
	}

	/**
	 * 角色分配数据权限
	 */
	@GetMapping("/authDataScope/{roleId}")
	public String authDataScope(@PathVariable("roleId") Long roleId, ModelMap mmap) {
		mmap.put("role", roleService.selectRoleById(roleId));
		return prefix + "/dataScope";
	}

	/**
	 * 保存角色分配数据权限
	 */
	@ResponseBody
	@PostMapping("/authDataScope")
	@RequiresPermissions("system:role:edit")
	@Log(title = "角色管理", businessType = BusinessType.UPDATE)
	public void authDataScopeSave(SysRole role) {
		roleService.checkRoleAllowed(role);
		role.setUpdateBy(ShiroUtils.getLoginName());
		if (roleService.authDataScope(role) > 0) {
			// ShiroUtils.setSysUser(userService.selectUserById(ShiroUtils.getSysUser().getUserId()));
		}
		// return error();
	}

	@ResponseBody
	@PostMapping("/remove")
	@RequiresPermissions("system:role:remove")
	@Log(title = "角色管理", businessType = BusinessType.DELETE)
	public void remove(String ids) throws Exception {
		roleService.deleteRoleByIds(ids);
	}

	/**
	 * 校验角色名称
	 */
	@ResponseBody
	@PostMapping("/checkRoleNameUnique")
	public int checkRoleNameUnique(SysRole role) {
		return roleService.checkRoleNameUnique(role);
	}

	/**
	 * 校验角色权限
	 */
	@ResponseBody
	@PostMapping("/checkRoleKeyUnique")
	public int checkRoleKeyUnique(SysRole role) {
		return roleService.checkRoleKeyUnique(role);
	}

	/**
	 * 选择菜单树
	 */
	@GetMapping("/selectMenuTree")
	public String selectMenuTree() {
		return prefix + "/tree";
	}

	/**
	 * 角色状态修改
	 */
	@ResponseBody
	@PostMapping("/changeStatus")
	@RequiresPermissions("system:role:edit")
	@Log(title = "角色管理", businessType = BusinessType.UPDATE)
	public void changeStatus(SysRole role) {
		roleService.checkRoleAllowed(role);
		roleService.changeStatus(role);
	}

	/**
	 * 分配用户
	 */
	@RequiresPermissions("system:role:edit")
	@GetMapping("/authUser/{roleId}")
	public String authUser(@PathVariable("roleId") Long roleId, ModelMap mmap) {
		mmap.put("role", roleService.selectRoleById(roleId));
		return prefix + "/authUser";
	}

	/**
	 * 查询已分配用户角色列表
	 */
	@ResponseBody
	@PostMapping("/authUser/allocatedList")
	@RequiresPermissions("system:role:list")
	public TableDataInfo<SysUser> allocatedList(SysUser user) {
		startPage();
		List<SysUser> list = userService.selectAllocatedList(user);
		return getDataTable(list);
	}

	/**
	 * 取消授权
	 */
	@ResponseBody
	@PostMapping("/authUser/cancel")
	@Log(title = "角色管理", businessType = BusinessType.GRANT)
	public void cancelAuthUser(SysUserRole userRole) {
		roleService.deleteAuthUser(userRole);
	}

	/**
	 * 批量取消授权
	 */
	@ResponseBody
	@PostMapping("/authUser/cancelAll")
	@Log(title = "角色管理", businessType = BusinessType.GRANT)
	public void cancelAuthUserAll(Long roleId, String userIds) {
		roleService.deleteAuthUsers(roleId, userIds);
	}

	/**
	 * 选择用户
	 */
	@GetMapping("/authUser/selectUser/{roleId}")
	public String selectUser(@PathVariable("roleId") Long roleId, ModelMap mmap) {
		mmap.put("role", roleService.selectRoleById(roleId));
		return prefix + "/selectUser";
	}

	/**
	 * 查询未分配用户角色列表
	 */
	@ResponseBody
	@PostMapping("/authUser/unallocatedList")
	@RequiresPermissions("system:role:list")
	public TableDataInfo<SysUser> unallocatedList(SysUser user) {
		startPage();
		List<SysUser> list = userService.selectUnallocatedList(user);
		return getDataTable(list);
	}

	/**
	 * 批量选择用户授权
	 */
	@ResponseBody
	@PostMapping("/authUser/selectAll")
	@Log(title = "角色管理", businessType = BusinessType.GRANT)
	public void selectAuthUserAll(Long roleId, String userIds) {
		roleService.insertAuthUsers(roleId, userIds);
	}
}